package 快排;

import java.util.PriorityQueue;

/**
 * @Date 2024/10/8 15:36
 * @description: 第 k 小
 * .
 * @Author LittleNight
 */
public class likou159 {

    // 创建大跟堆
    public int[] inventoryManagement(int[] nums, int k) {
        // 创建大跟堆
        PriorityQueue<Integer> heap = new PriorityQueue<>((a, b) -> b - a);
        for(int x : nums) {
            heap.offer(x);
            if(heap.size() > k) heap.poll();
        }
        // 堆顶元素即为第k小的元素.
        int[] ret = new int[k];
        int i = 0;
        while(!heap.isEmpty()) {
            ret[i++] = heap.poll();
        }
        return ret;
    }
}
